/* See LICENSE file for copyright and license details. */
#include <ifaddrs.h>
#include <netdb.h>
#include <stdio.h>
#include <string.h>
#include <dirent.h>
#if defined(__OpenBSD__)
	#include <sys/socket.h>
	#include <sys/types.h>
#elif defined(__FreeBSD__)
	#include <netinet/in.h>
	#include <sys/socket.h>
#endif

#include "../slstatus.h"
#include "../util.h"

static const char *
ip(const char *interface, unsigned short sa_family)
{
	struct ifaddrs *ifaddr, *ifa;
	int s;
	char host[NI_MAXHOST];

	if (getifaddrs(&ifaddr) < 0) {
		warn("getifaddrs:");
		return NULL;
	}

	for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
		if (!ifa->ifa_addr)
			continue;

		s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in6),
		                host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
		if (!strcmp(ifa->ifa_name, interface) &&
		    (ifa->ifa_addr->sa_family == sa_family)) {
			freeifaddrs(ifaddr);
			if (s != 0) {
				warn("getnameinfo: %s", gai_strerror(s));
				return NULL;
			}
			return bprintf("%s", host);
		}
	}

	freeifaddrs(ifaddr);

	return NULL;
}

const char *
ipv4(const char *interface)
{
	return ip(interface, AF_INET);
}

const char *
ipv6(const char *interface)
{
	return ip(interface, AF_INET6);
}

const char *
ipauto()
{
	struct ifaddrs *ifaddr, *ifa;
	int s;
	char host[NI_MAXHOST];
	DIR *dp;
	struct dirent *ep;

	dp = opendir("/sys/class/net");
	if (dp != NULL) {
	    while ((ep = readdir(dp))) {
		char *iface = ep->d_name;
		// skip always existing dirs and loopback device
		if (strcmp(iface, ".") == 0 || strcmp(iface, "..") == 0 || strcmp(iface, "lo") == 0)
		    continue;
		if (getifaddrs(&ifaddr) == -1) {
			warn("Failed to get IPv4 address for interface %s", iface);
			//return NULL;
			continue;
		}

		for (ifa = ifaddr; ifa != NULL; ifa = ifa->ifa_next) {
			if (ifa->ifa_addr == NULL) {
				continue;
			}
			s = getnameinfo(ifa->ifa_addr, sizeof(struct sockaddr_in), host, NI_MAXHOST, NULL, 0, NI_NUMERICHOST);
			if ((strcmp(ifa->ifa_name, iface) == 0) && (ifa->ifa_addr->sa_family == AF_INET)) {
				if (s != 0) {
					warnx("Failed to get IPv4 address for interface %s", iface);
					//return NULL;
					continue;
				}
				return bprintf("%s", host);
			}
		}
	    }
	    (void) closedir (dp);
	}
	else {
	    warnx("Failed to get IPv4 address because /sys/class/net cannot be enumerated");
	}

	freeifaddrs(ifaddr);

	return bprintf("%s", "none");
}

